//
// Created by Tony on 2024/8/3.
//

#include "ClassModule.h"
#include "../../../../compiler/Visitor.h"

namespace Parser {

void ClassModule::addImport(shared_ptr<ImportStmt> import) {
    children.push_back(import);
    mImports.push_back(import);
}

vector<shared_ptr<ImportStmt>> ClassModule::getImports() { return mImports; }

void ClassModule::addClassDefine(shared_ptr<ClassDefine> classDefine) {
    children.push_back(classDefine);
    mClassDefines.push_back(classDefine);
}

vector<shared_ptr<ClassDefine>> ClassModule::getClassDefines() { return mClassDefines; }

int32_t ClassModule::accept(Compiler::Visitor* visitor) { return visitor->visit(this); }

} // namespace Parser